package com.boge.ai.entity.milvus;

/**
 * 通过Milvus向量数据库做RAG的实体对象
 */
public class MilvusEntity {
    /**
     * 向量数据库名称
     */
    public static final String DB_NAME = "default";

    /**
     * 集合名称
     */
    public static final String COLLECTION_NAME = "rag_boge";

    /**
     * 分片数量
     */
    public static final int SHARDS_NUM = 1;

    /**
     * 分区数量
     */
    public static final int PARTITION_NUM = 1;

    /**
     * 特征向量维度
     */
    public static final Integer FEATURE_DIM = 1536;

    /**
     * 字段
     */
    public static class Field {

        /**
         * id
         */
        public static final String ID = "id";

        /**
         * 文本特征向量
         */
        public static final String FEATURE = "feature";

        /**
         * 文本
         */
        public static final String INSTRUCTION = "instruction";

        /**
         * 问答匹配的结果
         */
        public static final String OUTPUT = "output";

    }
}

